import time
from common.login import Login
from common.report_generate import Report

class Charater:
    def __init__(self):
        self.st = Login()
        self.driver = self.st.driver
        self.insert = Report()

    #超级管理员
    def admin(self):
        self.st.login_fast('超级管理员')
        list = ['公司官网','培训资源','就业管理','员工信息','学员信息','市场营销','教师值班','申请加班','财务流水','学员缴费','领用登记','归还登记']
        try:
            for i in list:
                self.driver.find_element_by_link_text(i)
            print('ok')
            self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '成功', '无', '无')
            self.driver.quit()
        except:
            print('not ok')
            img = self.insert.pic_screenshot()
            self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '失败', '无', img)
            self.driver.quit()



    #咨询主管
    def advisory(self):
        self.st.login_fast('咨询主管')
        lists = ['公司官网','培训资源','就业管理','员工信息']

        try:
            for i in lists:
                self.driver.find_element_by_link_text(i)
                print(f'{i}ok')
            self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '成功', '无', '无')
            self.driver.quit()
        except:
           print(f'{i}not ok')
           img = self.insert.pic_screenshot()
           self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '失败', '无', img)
           self.driver.quit()

    #咨询师
    def consultative(self):
        self.st.login_fast('咨询师')
        try:
            self.driver.find_element_by_link_text('公司官网')
            self.driver.find_element_by_link_text('培训资源')
            self.driver.find_element_by_link_text('学员信息')
            self.driver.find_element_by_link_text('员工信息')
            self.driver.find_element_by_link_text('○    今日分配的资源')
            self.driver.find_element_by_link_text('○    待回访的资源')
            self.driver.find_element_by_link_text('○    今日将要被废弃的资源')
            print('ok')
            self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '成功', '无', '无')
        except:
            print('not ok')
            img = self.insert.pic_screenshot()
            self.insert.write_report('3.5', '首页', 'GUI测试', 'TC-013', '不同角色首页显示', '失败', '无', img)




if __name__ == '__main__':
    class1 = Charater()
    class1.advisory()